from transformers import AutoModelForCausalLM

# 使用 PyTorch 的模型打印方式，主要看模型的组成
model_path = "/root/ModelSpace/Qwen2.5-1.5B-Instruct"
model = AutoModelForCausalLM.from_pretrained(model_path, torch_dtype="auto", device_map="auto")
print(model)
